package com.xigua._树;

import com.xigua.TreeNode;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * @author LiRongHua
 * @Title: ${file_name}
 * @Package ${package_name}
 * @Description: 给你一个含重复值的二叉搜索树（BST）的根节点 root ，
 * 找出并返回 BST 中的所有 众数（即，出现频率最高的元素）。
 *如果树中有不止一个众数，可以按 任意顺序 返回。
 *
 * @date 2022/4/123:57
 */
public class _501_二叉搜索树中的众数 {

    public static void main(String[] args) {
        TreeNode root = new TreeNode(1,null,new TreeNode(2));
        findMode1(root);
    }













    static HashMap<Integer,Integer> map = new HashMap<>();
    public static int[] findMode1(TreeNode root) {
        dfs(root);
        List<Integer> list = new ArrayList<>();
        int max = 0;
        for (Map.Entry<Integer, Integer> entry : map.entrySet()) {
            if (entry.getValue()>max){
                max= entry.getValue();
                list.clear();
                list.add(entry.getKey());
            }else if (entry.getValue() == max){
                list.add(entry.getKey());
            }
        }
        int ans[] = new int[list.size()];
        for (int i = 0; i < ans.length; i++) {
            ans[i] = list.get(i);
        }
        return ans;
    }

    public static void dfs(TreeNode root){
        if (root==null) return;
        dfs(root.left);
        map.put(root.val,map.getOrDefault(root.val,0)+1);
        dfs(root.right);
    }


}
